import os
from utils import FileUtils

def generate_board_heat_trend():
    """生成板块热力趋势图"""
    # 获取当前文件所在目录
    current_dir = os.path.dirname(os.path.abspath(__file__))
    
    # 设置数据目录和输出文件路径
    data_dir = os.path.join(current_dir, "data")
    html_path = os.path.join(current_dir, "board_heat_trend.html")
    
    # 设置颜色方案：从深蓝色(0)到正红色(100)
    colorscale = [
        [0, 'rgb(0, 0, 128)'],      # 深蓝色
        [0.25, 'rgb(0, 100, 255)'],  # 蓝色
        [0.5, 'rgb(255, 255, 255)'], # 白色
        [0.75, 'rgb(255, 100, 0)'],  # 橙色
        [1, 'rgb(255, 0, 0)']        # 正红色
    ]
    
    # 生成热力图HTML
    html_file = FileUtils.generate_heatmap_html(data_dir, html_path, colorscale=colorscale)
    
    if html_file:
        print(f"热力图已生成: {html_file}")
    else:
        print("热力图生成失败")

if __name__ == "__main__":
    generate_board_heat_trend()